首页 探索 帮助
注册 登录
J1X
/
huginn
关注 2
点赞 0
派生 0

浏览代码
added home time line option to user agent

Judy Ngai 9 年之前
父节点
0116ac3d7d
当前提交
001c5c41ca
共有 1 个文件被更改，包括 37 次插入 和 11 次删除
显示文件统计
  1. 37 11
      app/models/agents/twitter_user_agent.rb

+ 37 - 11
app/models/agents/twitter_user_agent.rb
查看文件 

@@ -5,13 +5,19 @@ module Agents
5 5 
     cannot_receive_events!
6 6
7 7 
     description <<-MD
8 
-      The Twitter User Agent follows the timeline of a specified Twitter user.
8 
+      There are two options in using the Twitter User Agent.
9 
+
10 
+      The first option is to follow the timeline of a specified Twitter user.
11 
+
12 
+      The second option is to follow your own home timeline including both your tweets and from people whom you are following.
9 13
10 14 
       #{twitter_dependencies_missing if dependencies_missing?}
11 15
12 16 
       To be able to use this Agent you need to authenticate with Twitter in the [Services](/services) section first.
13 17
14 
-      You must also provide the `username` of the Twitter user to monitor.
18 
+      For the first option, you must provide the `username` of the Twitter user to monitor.
19 
+
20 
+      For the second option, you must remove 'username' and set 'choose_home_time_line' to 'true'.
15 21
16 22 
       Set `include_retweets` to `false` to not include retweets (default: `true`)
17 23 
       
@@ -57,12 +63,12 @@ module Agents
57 63 
         'username' => 'tectonic',
58 64 
         'include_retweets' => 'true',
59 65 
         'exclude_replies' => 'false',
60 
-        'expected_update_period_in_days' => '2'
66 
+        'expected_update_period_in_days' => '2',
67 
+        'choose_home_time_line' => 'false'
61 68 
       }
62 69 
     end
63 70
64 71 
     def validate_options
65 
-      errors.add(:base, "username is required") unless options['username'].present?
66 72 
       errors.add(:base, "expected_update_period_in_days is required") unless options['expected_update_period_in_days'].present?
67 73
68 74 
       if options[:include_retweets].present? && !%w[true false].include?(options[:include_retweets])
 
@@ -72,6 +78,9 @@ module Agents
72 78 
       if options[:starting_at].present?
73 79 
         Time.parse(options[:starting_at]) rescue errors.add(:base, "Error parsing starting_at")
74 80 
       end
81 
+      # if options[:username].present?
82 
+      #   errors.add(:base, "username is required")
83 
+      # end
75 84 
     end
76 85
77 86 
     def starting_at
 
@@ -82,6 +91,10 @@ module Agents
82 91 
       end
83 92 
     end
84 93
94 
+    def choose_home_time_line?
95 
+      interpolated[:choose_home_time_line] != "false"
96 
+    end
97 
+
85 98 
     def include_retweets?
86 99 
       interpolated[:include_retweets] != "false"
87 100 
     end
 
@@ -95,18 +108,31 @@ module Agents
95 108 
       opts = {:count => 200, :include_rts => include_retweets?, :exclude_replies => exclude_replies?, :include_entities => true, :contributor_details => true}
96 109 
       opts.merge! :since_id => since_id unless since_id.nil?
97 110
111 
+      if choose_home_time_line?
112 
+
98 113 
       # http://rdoc.info/gems/twitter/Twitter/REST/Timelines#user_timeline-instance_method
99 
-      tweets = twitter.user_timeline(interpolated['username'], opts)
114 
+        tweets = twitter.home_timeline(opts)
100 115
101 
-      tweets.each do |tweet|
102 
-        if tweet.created_at >= starting_at
103 
-          memory['since_id'] = tweet.id if !memory['since_id'] || (tweet.id > memory['since_id'])
116 
+        tweets.each do |tweet|
117 
+          if tweet.created_at >= starting_at
118 
+            memory['since_id'] = tweet.id if !memory['since_id'] || (tweet.id > memory['since_id'])
104 119
105 
-          create_event :payload => tweet.attrs
120 
+            create_event :payload => tweet.attrs
121 
+          end
106 122 
         end
107 
-      end
108 123
109 
-      save!
124 
+      else
125 
+        tweets = twitter.user_timeline(interpolated['username'], opts)
126 
+
127 
+        tweets.each do |tweet|
128 
+          if tweet.created_at >= starting_at
129 
+            memory['since_id'] = tweet.id if !memory['since_id'] || (tweet.id > memory['since_id'])
130 
+
131 
+            create_event :payload => tweet.attrs
132 
+          end
133 
+        end
134 
+      #save!
135 
+      end
110 136 
     end
111 137 
   end
112 138 
 end